import torchvision
import torch.nn as nn

vgg16 = torchvision.models.vgg16(pretrained=True)
print(vgg16)
vgg16.classifier.add_module("add_linear",nn.Linear(1000,10))
print(vgg16)
vgg16.classifier[6] = nn.Linear(1000,10)

